hacker rank solutioin in python